Trying to fix a bug

Not done :(