Added a fix for a fail case I had