fix annoying bug