Fixes #6, small bug