Fixed return statement in genetic models