Bug with last commit

Fixed it