fixed bug again

in a better way