fix puzzle JS