Fixes broken test suite (why was it broken?)