Bug fix, thanks kevingranade