fix issue with wrong unit treating