Fixed the orientation reset in the GENERATE_PIECE state