fixes test file function return type