Fixes last commit which had small error