Fixed issue with assertion added to wrong method.