Refactor: cosmetic fix