Fix code, for spec