I know I'll need it in the next time so it wouldn't make sense to change that and yeah will add a quick unittest for that. This was actually needed for the webinar today so I first had to get this done :)