Fixed small bug introduced by last commit