Fixed small error in refactoring #53