Fix code I pushed by accident