Fixes broken test
Bug fix