small tweak to stop crash