Correct axis scale logic

Issue: #38