Remove if statement around the return