Fix orientation bug