Fix expression return annotation