Fix issue where assembly check wasn't happening