Fixed whitespace + small bug