fix type on return type