UI fix
Merge pull request #54 from ssirushchyshyn/develop

Small UI fix