Removed unused variable and updated feature test