The movement is fine as is! The fact it's so limiting forces you to think various steps ahead so as to not fall into checkmate situations, and that's pretty cool.
