Fixed small bag in code style