Removing extra return