3.1.1a

- Bug fix
- Mini refactor