Fixed small bug created by last commit