Minor bug fix

- [ ] @be-dmitry 
- [ ] @mikeklein13