fixed previous commit .. remember to run tests locally before pushing :)