Assertions were in the wrong spot

Moved the assertions down to where they should have been