fixed bug that failed the test