Fixing mistake form previous commit