Fixed small bug - tests green.