added return assertion