quick fixed

still not sure if it works