fixes #2
fixing bug