Fixes test class.