[Closes #54] Fixed minor bug