Hopefully fixed a test