Fixed assertionError