fix bug

Hi, it should work now