Fixing another problem
