Fix import that prevented tests from running.  References #7