Fixed error that was failing the cucumber tests