Fixed my own bug