Trying to fix one bug