minor fix, beautified code