Removed unnecessary assertion and import statements