Fixes bug with multiple tests being run