Fixed bug with comments after statements