fixes #1 fixs the bug