Fix the test :D thanks @mbabker