Fixing little code smell