Wrong function return type fixed