Fixed a small bug caused by fixing a small bug