Fix return type printing