Fix for issue #3

Rather simple.