Fix return bug that wasn't a bug