Fixed return test