Fixed a small bug to make things compile right