small pull request
fixed the first bug :P