Small function compile fix