Fixed the function