Moved the starting square assert

This assert proved to be very impractical during tests, so I moved it to
the method where it is actually relevant.