Fix ensure issue