Fixed issue with code style