Horrible hack fixes bug