Removing the extra return