Small change after code cleanup