Fixed tests for me.