Try fixing test

Code Style