Change >>= return to <$>