Fix integration tests for edx solutions

Great! Much better than a simple wait, it should save us from some flakiness down the line : )