There's a bug with the last "puzzle" where if you rewind twice in a row without moving the platform, the platform loses proper track of its current location. You can restart the level in the pause menu to reset it if you encounter this.