Fixed wierd bug