Simple test case and a fix for the original issue
Imports