Small function definition correction