Issue: #240 - Fixing bug