FIX: handle non-test files