I think I needed to add buffer for the player before loosing the wall jump. For wall jumps the trick is to jump and then move.