Fix return type validation derp